# 不仅查询当前城市的美食，还对美食进行简单介绍
import os

from langchain import PromptTemplate
from langchain.chains import LLMChain, SimpleSequentialChain
from langchain.llms import OpenAI

os.environ["OPENAI_API_KEY"] = '读者申请的OpenAI API Key'
cate_template = "你现在是一位美食博主，需要根据用户输入的城市：{city}，给出对应的美食推荐"
cate_prompt = PromptTemplate(template=cate_template, input_variables=['city'])
llm = OpenAI(temperature=0)
# 第一个链
cate_chain = LLMChain(llm=llm, prompt=cate_prompt)
introduction_template = "你现在是一位美食博主，需要根据用户输入的美食信息：{cate}，给出对应的美食简单介绍"
introduction_prompt = PromptTemplate(template=introduction_template, input_variables=['cate'])
# 第二个链
introduction_chain = LLMChain(llm=llm, prompt=introduction_prompt)
# 顺序链
result_chain = SimpleSequentialChain(chains=[cate_chain, introduction_chain], verbose=True)
result_chain.run("成都")
